import java.io.*;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;

// 扫描指定目录，并找到名称或者内容中包含指定字符的所有普通文件（不包含目录）
// 方案性能较差，所以尽量不要在太复杂的目录下或者大文件下实验, 遇到大文件内存不够用.
public class IODemo14 {
    public static void main(String[] args) throws IOException {
        Scanner scanner = new Scanner(System.in);

        System.out.print("请输入要扫描的根目录(绝对路径 或 相对路径)：");
        String rootDirPath = scanner.next();
        File rootDir = new File(rootDirPath);
        if (!rootDir.isDirectory()) {
            System.out.println("您输入的根目录不存在或不是目录！");
            return;
        }

        System.out.print("请输入要找的字符：");

        String token = scanner.next();

        List<File> result = new ArrayList<>();

        scanDirWithContent(rootDir, token, result);

        System.out.println("以下 " + result.size() + " 个文件包含 \'" + token +"\'：");

        for (File file : result) {
            System.out.println(file.getCanonicalPath());
        }
    }
    private static void scanDirWithContent(File rootDir, String token, List<File> result) {
        File[] files = rootDir.listFiles();// 获取当前目录下的所有文件
        if (files == null) {
            return;
        }
        for (File file : files) {
            System.out.println("当前搜索到：" + file.getAbsolutePath());
            if (file.isDirectory()) {
                // 是目录
                scanDirWithContent(file, token, result);
            } else if (file.isFile()) {
                // 是普通文件
                if (file.getName().contains(token)) {
                    // 文件名是否包含
                    result.add(file);
                    continue;
                }
                if (isContentContains(file, token)) {
                    // 文件内容
                    result.add(file);
                }
            } else {
                // 不是目录和普通文件, 跳过
                continue;
            }
        }
    }
    // 使用字节流 搭配 Scanner
    private static boolean isContentContains(File file, String token) {
        StringBuilder stringBuilder = new StringBuilder();
        try (InputStream inputStream = new FileInputStream(file)) {
            try (Scanner scanner = new Scanner(inputStream, "utf-8")) {
                while (scanner.hasNextLine()) {
                    // 一次读一行
                    stringBuilder.append(scanner.nextLine());
                    stringBuilder.append("\r\n");
                }
            }
        } catch (IOException e) {
            e.printStackTrace();
        }

        return stringBuilder.indexOf(token) != -1;
    }
    // 使用字符流
    private static boolean isContentContains2(File file, String token) {
        StringBuilder stringBuilder = new StringBuilder();
        try (Reader reader = new FileReader(file)) {
            while (true) {
                int c = reader.read();
                if (c == -1) {
                    break;
                }
                stringBuilder.append((char)c);
            }
        } catch (IOException e) {
            e.printStackTrace();
        }

        return stringBuilder.indexOf(token) != -1;
    }
}
